from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSequenceClassification
from datasets import load_dataset

model = AutoModelForSequenceClassification.from_pretrained("/data1/home/yifeiwang/MPCFormer/src/main/checkpoints/mnli")

model.save_pretrained("/data1/home/yifeiwang/MPCFormer/src/main/checkpoints/mnli_bin", safe_serialization=False)